package com.mou.aiagent.deom.invoke;

import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.chat.messages.AssistantMessage;
import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

/**
 * Spiing Ai 调用 测试实例
 */
@Component
@Slf4j
public class SpringAiAiInvoke implements CommandLineRunner {

    @Resource
    private ChatModel desktopChatModel;
    @Override
    public void run(String... args) throws Exception {
        // 调用desktopChatModel的call方法，传入一个Prompt对象，获取结果
        AssistantMessage output = desktopChatModel.call(
                 new Prompt("今天有什么特别的新闻吗？"))
                .getResult()
                .getOutput();
        // 打印输出结果
System.out.println(output.getText());


    }
}
